import unittest
import sys,  os
sys.path.append( os.getcwd() )
from matplotwrapper import *

class TestMatplotWrapper(unittest.TestCase):
  
  def setUp(self):    
    t = [1,3,6,10,20,40,70,100,200,400,700,1000,2000,3000]
    ti = []
    s = [3.4,4.2,4.8,5.08,5.60,6.05,6.40,6.65,7.10,7.50,7.90,8.25,8.70,8.90]
    for tt in t:
      ti.append( tt/24.0/60.0 )
    self.mpw = MatplotWrapper(ti, s)    

  def testGetRandomString(self):
    # make sure the shuffled sequence does not lose any elements
    print ( self.mpw.getRandomString() )

  def testCreateImageAll(self):
    print ( self.mpw.createImageAll() )   
    
if __name__ == '__main__':
    unittest.main()
